striker suffered his hamstring in the 2-1 win over stoke on april 4 .
costa has recovered ahead of schedule and will train with squad next week .
he could return for their top-of-the-table clash with arsenal next weekend .
jose mourinho 's side face manchester united on saturday .